package web.controllers;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.PrintWriter;
import javax.servlet.RequestDispatcher;
import javax.servlet.ServletConfig;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import modelovectorial.DocumentoPDF;
import web.utils.BrowserInformation;

/**
 * @project DLC Final / UTN FRC 2013
 * @date Febrero de 2013
 *
 * @author Gonzalo Arribere
 * @author Gustavo Morales
 * @author Pablo Moreno
 */
public class DownloadController extends HttpServlet
{
    private String mWorkDirectory;

    @Override
    public void init (ServletConfig config)
    {
        mWorkDirectory = config.getServletContext ().getRealPath ("/pdf").replace ("build\\web\\", "");
    }
    protected void processRequest(HttpServletRequest request, HttpServletResponse response)
    throws ServletException, IOException
    {
        int pIdDocument = 0;
        String strIdDocument = request.getParameter ("id");
        if (strIdDocument != null)
        {
            try
            {
                pIdDocument = Integer.parseInt (strIdDocument);
            }
            catch (Exception e)
            {
                pIdDocument = 0;
            }
        }
        if (pIdDocument > 0)
        {
            DocumentoPDF documento = DocumentoPDF.getDocument (pIdDocument);

            if (documento != null)
            {
                sendFile (documento, response);
            }
            else
            {
                response.setContentType("text/html;charset=UTF-8");
                RequestDispatcher dispatcher = request.getRequestDispatcher
                        ((BrowserInformation.isMovileBrowser (request)) ? SitePages.MOVILE_VIEW_NOT_FOUND : SitePages.PAGE_VIEW_NOT_FOUND);
                dispatcher.forward(request, response);
            }

        }
        else
        {
            response.setContentType("text/html;charset=UTF-8");
            RequestDispatcher dispatcher = request.getRequestDispatcher
                    ((BrowserInformation.isMovileBrowser (request)) ? SitePages.MOVILE_VIEW_NOT_FOUND : SitePages.PAGE_VIEW_NOT_FOUND);
            dispatcher.forward(request, response);
        }
    }
    private void sendFile (DocumentoPDF pDocumento, HttpServletResponse pResponse) throws IOException
    {
        PrintWriter out = pResponse.getWriter();

        try
        {
            pResponse.setContentType ("application/pdf");

            String disHeader = "Content-Disposition: attachment; filename=\"" + pDocumento.getRuta () + "\"";

            pResponse.setHeader ("Content-Disposition", disHeader);

            // transfer the file byte-by-byte to the response object
            File fileToDownload = new File (mWorkDirectory + File.separator + pDocumento.getRuta ());
            FileInputStream fileInputStream = new FileInputStream (fileToDownload);

            int i;
            while ((i = fileInputStream.read ()) != -1)
            {
                out.write (i);
            }
            fileInputStream.close ();
            out.close ();
        }
        catch (Exception e) // file IO errors
        {
            e.printStackTrace ();
        }
    }

    // <editor-fold defaultstate="collapsed" desc="HttpServlet methods. Click on the + sign on the left to edit the code.">
    /** 
     * Handles the HTTP <code>GET</code> method.
     * @param request servlet request
     * @param response servlet response
     * @throws ServletException if a servlet-specific error occurs
     * @throws IOException if an I/O error occurs
     */
    @Override
    protected void doGet (HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException
    {
        processRequest (request, response);
    }

    /** 
     * Handles the HTTP <code>POST</code> method.
     * @param request servlet request
     * @param response servlet response
     * @throws ServletException if a servlet-specific error occurs
     * @throws IOException if an I/O error occurs
     */
    @Override
    protected void doPost (HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException
    {
        processRequest (request, response);
    }

    /** 
     * Returns a short description of the servlet.
     * @return a String containing servlet description
     */
    @Override
    public String getServletInfo ()
    {
        return "Short description";
    }// </editor-fold>


}
